package redis

import (
	"context"
	"fmt"
	"github.com/go-redis/redis/v8"
	"shortvideo2zk/common/nacos"
	"time"
)

var Rdb *redis.Client

func InitRedis() {
	nac := nacos.NacosConfig.Redis
	Rdb = redis.NewClient(&redis.Options{
		Addr:     nac.Addr,
		Password: nac.Password, // no password set
		DB:       nac.Db,       // use default DB
	})
	err := Rdb.Ping(context.Background()).Err()
	if err != nil {
		panic("redis连接失败")
	}
	fmt.Println("redis连接成功")
}

func SetRedis(key string, code int) {
	Rdb.Set(context.Background(), key, code, time.Minute*10)
}

func GetRedis(key string) int {
	i, err := Rdb.Get(context.Background(), key).Int()
	if err != nil {
		return 0
	}
	return i
}
